#include "problems/215.h"
#include "gtest/gtest.h"

TEST(TestProblemSolutions, TestProblem215)
{
    problem_215::Solution s;
    vector<int> nums{3, 2, 1, 5, 6, 4};
    ASSERT_EQ(s.findKthLargest(nums, 2), 5);
    nums = {3, 2, 3, 1, 2, 4, 5, 5, 6};
    ASSERT_EQ(s.findKthLargest(nums, 4), 4);
}